import pymysql
from stockholder import settings
import logging
from logging import handlers


class DbUtils:
    def __init__(self):
        pass

    def opencon(self):
        con = pymysql.connect(
            host=settings.MYSQL_HOST,
            db=settings.MYSQL_DBNAME,
            user=settings.MYSQL_USER,
            passwd=settings.MYSQL_PASSWD,
            charset='utf8',
            use_unicode=True)
        return con

    # 创建数据库
    def createdb(self):
        conn = self.opencon()
        sql = "create database if not exists " + self.db
        cur = conn.cursor()
        cur.execute(sql)
        cur.close()
        conn.close()

    # 执行sql
    def executesql(self, sql):
        conn = self.opencon()
        cur = conn.cursor()
        cur.execute(sql)
        conn.commit()
        # cur.close()
        # conn.close()

    # 插入数据
    def insert(self, sql, *params):
        conn = self.opencon()
        cur = conn.cursor()
        cur.execute(sql, params)
        conn.commit()
        cur.close()
        # conn.close()

    # 更新数据
    def update(self, sql, *params):
        conn = self.opencon()

        cur = conn.cursor()
        cur.execute(sql, params)
        conn.commit()
        cur.close()
        # conn.close()

    # 删除数据
    def delete(self, sql, *params):
        conn = self.opencon()

        cur = conn.cursor()
        cur.execute(sql, params)
        conn.commit()
        cur.close()
        # conn.close()


class Logger(object):
    level_relations = {
        'debug': logging.DEBUG,
        'info': logging.INFO,
        'warning': logging.WARNING,
        'error': logging.ERROR,
        'crit': logging.CRITICAL
    }  # 日志级别关系映射

    def __init__(self, filename, level='info', when='D', backCount=3,
                 fmt='%(asctime)s - %(pathname)s[line:%(lineno)d] - %(levelname)s: %(message)s'):
        self.logger = logging.getLogger(filename)
        format_str = logging.Formatter(fmt)  # 设置日志格式
        self.logger.setLevel(self.level_relations.get(level))  # 设置日志级别
        sh = logging.StreamHandler()  # 往屏幕上输出
        sh.setFormatter(format_str)  # 设置屏幕上显示的格式
        th = handlers.TimedRotatingFileHandler(filename=filename, when=when, backupCount=backCount,
                                               encoding='utf-8')  # 往文件里写入#指定间隔时间自动生成文件的处理器
        # 实例化TimedRotatingFileHandler
        # interval是时间间隔，backupCount是备份文件的个数，如果超过这个个数，就会自动删除，when是间隔的时间单位，单位有以下几种：
        # S 秒
        # M 分
        # H 小时、
        # D 天、
        # W 每星期（interval==0时代表星期一）
        # midnight 每天凌晨
        th.setFormatter(format_str)  # 设置文件里写入的格式
        self.logger.addHandler(sh)  # 把对象加到logger里
        self.logger.addHandler(th)


if __name__ == '__main__':
    log = Logger('all.log', level='debug')
    log.logger.debug('debug')
    log.logger.info('info')
    log.logger.warning('警告')
    log.logger.error('报错')
    log.logger.critical('严重')
    Logger('error.log', level='error').logger.error('error')
